package com.zy.explore.user.interfaces.assembler;

import com.zy.explore.common.interfaces.frm.CommonIdFrm;
import com.zy.explore.common.interfaces.frm.CommonIdsFrm;
import com.zy.explore.common.infrastructure.util.DataBeanUtil;
import com.zy.explore.common.interfaces.vo.PageVO;

import com.zy.explore.user.domain.jpa.entity.User;
import com.zy.explore.user.infrastructure.exception.MessageCode;
import com.zy.explore.user.interfaces.frm.UserFrm;
import com.zy.explore.user.interfaces.vo.UserVO;

import org.springframework.data.domain.Sort;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.util.CollectionUtils;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * (表t_user)数据转换辅助对象
 *
 * @author 中国印
 * @date 2021-09-20 14:36:06
 */
@Component
public class UserAssembler {

    /**
     * 新增数据
     *
     * @param frm 待新增的frm对象
     * @return 成功返回新增后的VO对象，失败返回null
     */
    public UserVO insert(UserFrm frm) {
        User entity = convertFromFrm(frm);
        entity.setCreateTime(new Date());
        entity.setUpdateTime(new Date());
        try {
            entity = entity.save();
        } catch (Exception e) {
            throw MessageCode.USER_NAME_REPEAT_ERROR.toException();
        }
        return convertToVO(entity);
    }

    /**
     * 删除数据
     *
     * @param frm 主键frm对象
     */
    public Integer delete(CommonIdFrm frm) {
        User entity = User.builder().id(Integer.parseInt(frm.getId())).build();
        entity.remove();
        return 1;
    }

    /**
     * 更新数据
     *
     * @param frm 待更新的frm对象
     * @return 成功返回更新后的VO对象，失败返回null
     */
    public UserVO update(UserFrm frm) {
        User entity = UserAssembler.convertFromFrm(frm);
        entity.setUpdateTime(new Date());
        return convertToVO(entity.updateSelective());
    }

    /**
     * 通过主键查询单条数据
     *
     * @param frm 主键Frm对象
     * @return 成功返回对应ID的VO对象，失败返回null
     */
    public UserVO findById(CommonIdFrm frm) {
        User entity = User.builder().id(Integer.parseInt(frm.getId())).build();
        return convertToVO(entity.findById());
    }

    /**
     * 通过主键批量查询数据集合
     *
     * @param frm 主键集合Frm对象
     * @return 成功返回对应ID的VO对象，失败返回空集合
     */
    public List<UserVO> findByIds(CommonIdsFrm frm) {
        List<UserVO> retList = new ArrayList<>();
        List<User> list = User.builder().build().findByIds(frm.getIds().stream().map(Integer::parseInt).collect(Collectors.toList()));
        if (!CollectionUtils.isEmpty(list)) {
            list.forEach(x -> retList.add(convertToVO(x)));
        }
        return retList;
    }

    /**
     * 根据条件获取数据集
     *
     * @param frm 查询条件对象
     * @return 成功返回满足条件的数据集合，失败返回空集合
     */
    public List<UserVO> getList(UserFrm frm) {
        List<UserVO> retList = new ArrayList<>();
        // 转换查询条件
        Specification<User> specification = User.getSpecification(frm);
        // 转换排序条件
        Sort sort = User.getSort(frm.getSortFields());
        User entity = User.builder().build();
        List<User> list = entity.findAll(specification, sort);
        if (!CollectionUtils.isEmpty(list)) {
            list.forEach(x -> retList.add(convertToVO(x)));
        }
        return retList;

    }

    /**
     * 根据条件查询数据分页
     *
     * @param frm 查询条件参数
     * @return 成功返回满足条件的分页，失败返回空页
     */
    public PageVO<UserVO> getPageInfo(UserFrm frm) {
        // 转换查询条件
        Specification<User> specification = User.getSpecification(frm);
        // 转换排序条件
        Sort sort = User.getSort(frm.getSortFields());
        User entity = User.builder().build();
        Page<User> page = entity.findAll(specification, frm.getPageNum(), frm.getPageSize(), sort);
        Page<UserVO> retPage = new PageImpl<>(new ArrayList<>(), page.getPageable(), page.getTotalElements());
        if (page.hasContent()) {
            List<UserVO> voList = new ArrayList<>();
            page.getContent().forEach(x -> voList.add(convertToVO(x)));
            retPage = new PageImpl<>(voList, page.getPageable(), page.getTotalElements());
        }
        return new PageVO<>(retPage);
    }


    /**
     * 实体类转VO
     *
     * @param entity 实体类
     * @return 转换后的对象
     */
    public static UserVO convertToVO(User entity) {
        return entity == null ? null : DataBeanUtil.copyOf(entity, UserVO.class, true);
    }

    /**
     * Frm转实体类
     *
     * @param frm 参数对象
     * @return 转换后的实体类对象
     */
    public static User convertFromFrm(UserFrm frm) {
        return DataBeanUtil.copyOf(frm, User.class, true);
    }
}

